
#ifndef _FUNCTION_HPP
#define _FUNCTION_HPP
#include <iostream>
#include "LightMatch.hpp"
#include <cmath>
#include <string>
#include <torch/script.h>
#include <iostream>
#include <memory>
#include <string>

class Function
{

public:
    std::vector<aim_deps::Light> lights;
    void findlight(cv::Mat frame, torch::jit::script::Module module);
    float pointDist(cv::Point2f a, cv::Point2f b);
};
#endif
